NoneType' object has no attribute 'shape